package mysql

import (
	"fmt"
	"leeframe/configs"

	"gorm.io/driver/mysql"
	"gorm.io/gorm"
	"gorm.io/gorm/schema"
)

var db *gorm.DB

type Model struct {
	ID int `gorm:"primary_key" json:"id"`
}

// InitMysql Setup initializes the database instance
func InitMysql() error {
	var err error

	// "user:pass@tcp(127.0.0.1:3306)/dbname?charset=utf8mb4&parseTime=True&loc=Local"
	dsn := fmt.Sprintf("%s:%s@tcp(%s)/%s?charset=utf8&parseTime=True&loc=Local",
		configs.Get().Mysql.User,
		configs.Get().Mysql.Password,
		configs.Get().Mysql.Host,
		configs.Get().Mysql.Name)

	db, err = gorm.Open(mysql.Open(dsn), &gorm.Config{
		NamingStrategy: schema.NamingStrategy{
			TablePrefix:   configs.Get().Mysql.TablePrefix, // 表名前缀
			SingularTable: true,                            // 使用单数表名，启用该选项后，`User` 表将是`user`
		},
	})
	if err != nil {
		return err
	}

	sqlDB, err := db.DB()
	if err != nil {
		return err
	}

	// SetMaxIdleConns 设置空闲连接池中连接的最大数量
	sqlDB.SetMaxIdleConns(configs.Get().Mysql.MaxIdle)

	// SetMaxOpenConns 设置打开数据库连接的最大数量。
	sqlDB.SetMaxOpenConns(configs.Get().Mysql.MaxActive)

	return nil
}

// CloseDB close database connection
func CloseDB() error {
	sqlDB, err := db.DB()
	if err != nil {
		return err
	}

	return sqlDB.Close()
}

// GetDB get database client
func GetDB() *gorm.DB {
	return db
}

// Create create model
func Create(table interface{}) error {
	return db.Model(table).Create(table).Error
}
